// Copyright (c) 2024, BlockProject 3D

//

// All rights reserved.

//

// Redistribution and use in source and binary forms, with or without modification,

// are permitted provided that the following conditions are met:

//

//     * Redistributions of source code must retain the above copyright notice,

//       this list of conditions and the following disclaimer.

//     * Redistributions in binary form must reproduce the above copyright notice,

//       this list of conditions and the following disclaimer in the documentation

//       and/or other materials provided with the distribution.

//     * Neither the name of BlockProject 3D nor the names of its contributors

//       may be used to endorse or promote products derived from this software

//       without specific prior written permission.

//

// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use crate::codec_map_initializer;

use crate::compiler::util::imports::ProtocolStore;

use crate::compiler::Protocol;

use crate::gen::base::Error;

use crate::gen::codec::CodecMap;

use crate::gen::file::{Content, File, FileType};

use crate::gen::swift::imports::gen_imports;

use crate::gen::swift::message::{gen_message_decl, gen_message_from_slice_impl, gen_message_write_impl};

use crate::gen::swift::r#enum::gen_enum_decl;

use crate::gen::swift::solver::SwiftImportSolver;

use crate::gen::swift::structure::gen_structure_decl;

use crate::gen::swift::union::gen_union_decl;

use crate::gen::Generator;

use crate::gen::template::Template;

const TEMPLATE_CODEC_BASE: &[u8] = include_bytes!("./default_codec/base.template");

const TEMPLATE_CODEC_STRING: &[u8] = include_bytes!("./default_codec/string.template");

const TEMPLATE_CODEC_LIST: &[u8] = include_bytes!("./default_codec/list.template");

pub struct GeneratorSwift;

impl Generator for GeneratorSwift {

    type Error = Error;

    type Params<'a> = ProtocolStore<'a, SwiftImportSolver>;

    fn get_default_codecs<'fragment, 'variable>() -> CodecMap<'fragment, 'variable> {

        let mut codecs = CodecMap::new(codec_map_initializer! {

            "base" => TEMPLATE_CODEC_BASE

        });

        codecs.insert("string", Template::compile_with_includes(TEMPLATE_CODEC_STRING, &codecs).unwrap());

        codecs.insert("list", Template::compile_with_includes(TEMPLATE_CODEC_LIST, &codecs).unwrap());

        codecs

    }

    fn generate(

        proto: &Protocol,

        codec_map: &CodecMap,

        params: &ProtocolStore<SwiftImportSolver>,

    ) -> Result<Vec<File>, Self::Error> {

        let imports = gen_imports(params);

        let decl_structures = proto.structs.iter().map(|v| gen_structure_decl(proto, v));

        let decl_enums = proto.enums.iter().map(|v| gen_enum_decl(proto, v));

        let decl_messages_code = proto.messages.iter().map(|v| gen_message_decl(proto, codec_map, v));

        let impl_from_slice_messages_code =

            proto.messages.iter().map(|v| gen_message_from_slice_impl(proto, codec_map, v));

        let impl_write_messages_code = proto.messages.iter().map(|v| gen_message_write_impl(proto, codec_map, v));

        let decl_unions = proto.unions.iter().map(|v| gen_union_decl(proto, v));

        Ok(vec![

            File::new(

                FileType::Structure,

                format!("{}.structures.swift", proto.name()),

                Content::from_iter(decl_structures).header(&imports),

            ),

            File::new(

                FileType::Enum,

                format!("{}.enums.swift", proto.name()),

                &Content::from_iter(decl_enums),

            ),

            File::new(

                FileType::Message,

                format!("{}.messages.swift", proto.name()),

                Content::try_from_iter(decl_messages_code)?.header(&imports),

            ),

            File::new(

                FileType::MessageWriting,

                format!("{}.messages_write.swift", proto.name()),

                Content::try_from_iter(impl_write_messages_code)?.header(&imports),

            ),

            File::new(

                FileType::MessageReading,

                format!("{}.messages_from_bytes.swift", proto.name()),

                Content::try_from_iter(impl_from_slice_messages_code)?.header(&imports),

            ),

            File::new(

                FileType::Union,

                format!("{}.unions.swift", proto.name()),

                Content::from_iter(decl_unions).header(&imports),

            ),

        ])

    }

    fn get_language_extension() -> &'static str {

        "swift"

    }

}